package com.jcx.ldzj.config;

import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import com.jcx.ldzj.entity.LoginUser;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.reflection.MetaObject;
import org.apache.shiro.SecurityUtils;
import org.springframework.stereotype.Component;

import java.time.LocalDateTime;

/**
 * 公共字段填充
 *
 * @author linpz
 */
@Slf4j
@Component
public class CustomMetaObjectHandler implements MetaObjectHandler {
    /**
     * 插入元对象字段填充（用于插入时对公共字段的填充）
     *  在这里对属性进行插入的时候，哪怕数据库不存在该字段，依然不会报错，但对象中不存在该字段则会报错
     * @param metaObject 元对象
     */
    @Override
    public void insertFill(MetaObject metaObject) {
        LocalDateTime now = LocalDateTime.now();
        setFieldValByName("createTime", now, metaObject);
        setFieldValByName("updateTime", now, metaObject);
        if (metaObject.findProperty("state", true) != null) {
            Object o = metaObject.getValue("state") != null ? metaObject.getValue("state") : '1';
            setFieldValByName("state", o, metaObject);
        }
        LoginUser principal = (LoginUser) SecurityUtils.getSubject().getPrincipal();
        if(principal!=null){
            setFieldValByName("updataer", principal.getUserName(), metaObject);
            setFieldValByName("creatar", principal.getUserName(), metaObject);
        }
    }

    /**
     * 更新元对象字段填充（用于更新时对公共字段的填充）
     *
     * @param metaObject 元对象
     */
    @Override
    public void updateFill(MetaObject metaObject) {
        setFieldValByName("updateTime", LocalDateTime.now(), metaObject);
        LoginUser principal = (LoginUser) SecurityUtils.getSubject().getPrincipal();
        if(principal!=null){
            setFieldValByName("updataer", principal.getUserName(), metaObject);
        }
    }
}
